from models.yolo import ModelDistiller, Model

model = ModelDistiller("models/yolov5n_animal.yaml", "weights/yolov5l.pt", "configs/distill_config.py")
# model = Model("models/yolov5n_animal.yaml")
# print("-------------------------------------------------")
for x in model.named_modules():
    print(x)
